给定一个初始为空的栈和一系列压栈、弹栈操作,请编写程序输出每次弹栈的元素。栈的元素值均为整数。

输入格式:

输入第1行为1个正整数n,表示操作个数;接下来n行,每行表示一个操作,格式为1 d或0。1 d表示将整数d压栈,0表示弹栈。n不超过20000。

输出格式:

按顺序输出每次弹栈的元素,每个元素一行。若某弹栈操作不合法(如在栈空时弹栈),则对该操作输出invalid。

输入样例:

1
2
3
4
5
6
7
8
7
1 1
1 2
0
0
0
1 3
0

输出样例:

1
2
3
4
2
1
invalid
3

思路

c++ stl 中容器stack的基本使用,注意需要使用s.empty()来判断栈是否为空

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#include <iostream>
#include <stack>
using namespace std;
int main() {
stack<int> s;
int n;
cin >> n;
for (int i = 0; i < n; i++) {
int x;
cin >> x;
if (x) {
int y;
cin >> y;
s.push(y);
} else {
if (!s.empty()) {
cout << s.top() << endl;
s.pop();
} else
cout << "invalid" << endl;
}
}
return 0;
}